momentum space wave function